#include <pybind11/pybind11.h>

void hello_from_cpu(int n);

PYBIND11_MODULE(cpu_extension, m) {
    m.def("hello", &hello_from_cpu, "Print Hello from CPU multiple times");
}